CUDA_VISIBLE_DEVICES=2,3 python3 main.py
